package com.example.mq.mqserver.core;

import com.example.mq.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;


/*
*  这个类表示一个存储消息的队列
*  MSG => Message
* */
public class MSGQueue {

    // 表示队列的身份标识
    private String name;
    // 表示队列是否持久化， true 表示需要持久化， false表示不需要持久化
    private boolean durable = false;



    // 暂未实现的功能
    // 这个 独占 功能, 也是先把字段列在这里, 具体的独占功能暂时先不实现.
    // 这个属性为 true, 表示这个队列只能被一个消费者使用(别人用不了). 如果为 false 则是大家都能使用
    private boolean exclusive = false;
    // 这个 自动删除 功能, 也是先把字段列在这里, 具体的独占功能暂时先不实现.
    // 为 true 表示没有人使用之后, 就自动删除. false 则是不会自动删除.
    private boolean autoDelete = false;

    // 为了把这个 arguments 存到数据库中,  就需要把 Map 转成 json 格式的字符串.
    private Map<String , Object> arguments = new HashMap<>();// 也是表示扩展参数. 当前也是先列在这里, 先暂时不实现

    // 当前队列都有哪些消费者订阅了.
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();
    // 记录当前取到了第几个消费者. 方便实现轮询策略.
    private AtomicInteger consumerSeq = new AtomicInteger(0);


    // 添加一个新的订阅者
    public void addConsumerEnv(ConsumerEnv consumerEnv) {
        consumerEnvList.add(consumerEnv);
    }

    // 订阅者的删除暂时先不考虑.
    // 挑选一个订阅者, 用来处理当前的消息. (按照轮询的方式)
    public ConsumerEnv chooseConsumer() {
        if (consumerEnvList.size() == 0) {
            return null;
        }
        // 计算一下当前要取的元素的下标.
        int index = (consumerSeq.get() % consumerEnvList.size());
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);
    }



    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public boolean isDurable() {
        return durable;
    }

    public void setDurable(boolean durable) {
        this.durable = durable;
    }

    public boolean isExclusive() {
        return exclusive;
    }

    public void setExclusive(boolean exclusive) {
        this.exclusive = exclusive;
    }

    public boolean isAutoDelete() {
        return autoDelete;
    }

    public void setAutoDelete(boolean autoDelete) {
        this.autoDelete = autoDelete;
    }

    // 这里的 get set 方法 用于和数据库交互使用.
    public String getArguments(){ // 这个 getArguments() 方法 是用来把 arguments 数据转换成 字符串格式 存到数据库中
        // 是把当前的 arguments 参数, 从 Map 转成 String (JSON)
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        // 如果代码异常， 返回空的 json 字符串
        return "{}";
    }

    public void setArguments(String argumentsJson){ // 这个方法, 是从数据库读数据之后, 构造 Exchange 对象, 会自动调用到
        // 把参数中 argumentsJson 按照 json 格式解析，转换成 上面的 map 对象
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<HashMap<String, Object>>() {});
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

   public void setArguments(String key, Object value){
        arguments.put(key, value);
   }

    public void setArguments(Map<String, Object> arguments) {
        this.arguments = arguments;
    }

    public Object getArguments(String key){
        return arguments.get(key);
    }
}

